cd ../..

gpus=$1

python -m trainer.IntplContrastiveTrainer fit \
    --config configs/model/CRNN_contrastive.yaml \
    --config configs/dataset/desed_id_all.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive_ds_80k.yaml \
    --model.exp_name="RE/desed+CL/id+od_all/mask_ratio0.4_mask_len4/+mixup+0.5weak_bsz_192" \
    --data.init_args.batch_sizes=[12,12,12,192,96] \
    --trainer.devices=${gpus} 